package com.pig4cloud.pig.file.service.impl;

import cn.hutool.core.io.FileUtil;
import cn.hutool.core.io.IoUtil;
import cn.hutool.core.util.IdUtil;
import cn.hutool.core.util.StrUtil;
import com.amazonaws.services.s3.model.S3Object;
import com.pig4cloud.pig.common.core.util.R;
import com.pig4cloud.pig.common.file.core.FileProperties;
import com.pig4cloud.pig.common.file.core.FileTemplate;
import com.pig4cloud.pig.file.service.FarmFileService;
import jakarta.servlet.http.HttpServletResponse;
import lombok.AllArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
import org.springframework.web.multipart.MultipartFile;

import java.io.InputStream;
import java.time.LocalDate;
import java.util.HashMap;
import java.util.Map;


/**
 * 农业通用文件上传管理模块
 * @author: lty
 * @date: 2024/11/27 11:34
*/
@Slf4j
@Service
@AllArgsConstructor
public class FarmFileServiceImpl implements FarmFileService {

	//	因为加了构造函数注解，所以会构造函数自动注入
	private final FileTemplate fileTemplate;

	private final FileProperties properties;



	/**
	 * 用户访问文件的路径开头
	 */
	private final String fileAccessUrl="/admin/farmFile";

	/**
	 * 在bucketName下面的 相对保存目录
	 */
	private final String relativeDir="farmFile";


	/**
	 * 文件上传
	 * @param file
	 * @return
	 */
	@Override
	public R uploadFarmFile(MultipartFile file) {
		String suffixName = FileUtil.extName(file.getOriginalFilename());
		suffixName=handleFileSuffixName(suffixName);

		String fileName = IdUtil.simpleUUID() + StrUtil.DOT + suffixName;
		Map<String, Object> resultMap = new HashMap<>(4);
		String bucketNameThis = properties.getBucketName()+'/'+relativeDir; //todo 为了和整个系统的bucketName区分开，原有本地bucket加上一个标识用于存储什么类型的文件

		resultMap.put("bucketName", properties.getBucketName());
		resultMap.put("fileName", fileName);

		// 获取当前日期
		LocalDate currentDate = LocalDate.now();
		// 获取年、月、日
		int year = currentDate.getYear();
		int month = currentDate.getMonthValue();
		int day = currentDate.getDayOfMonth();
		// 拼接文件路径(带上年月日）
		String filePathDir = String.format("%s/%d/%02d/%02d",bucketNameThis, year, month, day);
		String fileUrl = String.format("%s/%s/%s",fileAccessUrl, filePathDir, fileName);
		resultMap.put("url", fileUrl);
		try (InputStream inputStream = file.getInputStream()) {
//			在 localtemplate即本地存储中 这个bucket理解为 文件相对路径就好 使用对象存储他就是存储而fileName才是 文件路径
			fileTemplate.putObject(filePathDir, fileName, inputStream, file.getContentType());
		}
		catch (Exception e) {
			log.error("上传失败", e);
			return R.failed(e.getLocalizedMessage());
		}
		return R.ok(resultMap);
	}


	/**
	 * 读取文件
	 * @param fileDir
	 * @param fileName
	 * @param response
	 */
	@Override
	public void getFile(String fileDir, String fileName, HttpServletResponse response) {
		try (S3Object s3Object = fileTemplate.getObject(fileDir, fileName)) {
//			response.setContentType("application/octet-stream; charset=UTF-8");
			// 获取文件的 MIME 类型
			String mimeType = getMimeType(fileName);
			response.setContentType(mimeType);
			// 设置 Content-Disposition 为 inline 以预览文件
			response.setHeader("Content-Disposition", "inline; filename=\"" + fileName + "\"");
			IoUtil.copy(s3Object.getObjectContent(), response.getOutputStream());
		}
		catch (Exception e) {
			log.error("文件读取异常: {}", e.getLocalizedMessage());
		}
	}




	private String handleFileSuffixName(String name){
		if(name.equalsIgnoreCase("jpeg") || name.equalsIgnoreCase("png") ){
			return name.toLowerCase();
		}
		if(name.equalsIgnoreCase("jpg")){ //同义转换为jpeg 方便yolo处理
			return "jpeg";
		}
		throw new RuntimeException("不支持的图片格式");
	}


	private String getMimeType(String fileName) {
		// 根据文件扩展名获取 MIME 类型
		if (fileName.endsWith(".jpg") || fileName.endsWith(".jpeg")) {
			return "image/jpeg";
		} else if (fileName.endsWith(".png")) {
			return "image/png";
		} else if (fileName.endsWith(".gif")) {
			return "image/gif";
		} else if (fileName.endsWith(".pdf")) {
			return "application/pdf";
		} else {
			return "application/octet-stream; charset=UTF-8";
		}
	}
}
